
# excel每一行数据的生成模板，默认会输出空值
# series ID
# group 
# 对应gt编号 
# 对应inf编号 
# 长径gt(mm) 
# 短径gt(mm) 
# 病灶面积gt(mm) 
# 最大结节密度gt  
# 平均结节密度gt  
# 长径inf(mm)  
# 短径inf(mm)  
# 病灶面积inf(mm)  
# 最大结节密度inf 
# 平均结节密度inf
# 分类仲裁结果
# 分类检测结果
# 预测检出概率
# 正常产品输出概率
# 最大结节密度
# 平均结节密度
# 检出统计结果
# 交并比
# 中心点距离

class ExcelTemp():
    def __init__(self):
        super(ExcelTemp, self).__init__()
        self.series_ID = None, 
        self.group = None,
        self.num_gt = None,
        self.num_inf = None,
        self.long_gt = None,
        self.short_gt = None,
        self.volume_gt = None,
        self.max_nodule_density_gt = None,
        self.mean_nodule_density_gt = None,
        self.long_inf = None,
        self.short_inf = None,
        self.volume_inf = None,
        self.max_nodule_density_inf = None,
        self.mean_nodule_density_inf = None,
        self.classification_results_gt = None,
        self.classification_results_inf = None,
        self.prob = None,
        self.normal_prob = None,
        self.max_nodule_density = None,
        self.mean_nodule_density = None,
        self.statistical_result = None,
        self.IoU = None,
        self.center_distance = None,

    def get_excel_temp(self):
        excel_temp = {}
        excel_temp['series ID'] = self.series_ID
        excel_temp['group'] = self.group
        excel_temp['对应gt编号'] = self.num_gt
        excel_temp['对应inf编号'] = self.num_inf
        excel_temp['长径gt(mm)'] = self.long_gt
        excel_temp['短径gt(mm)'] = self.short_gt
        excel_temp['病灶体积gt(mm)'] = self.volume_gt
        # excel_temp['最大结节密度gt'] = self.max_nodule_density_gt
        excel_temp['平均结节密度gt'] = self.mean_nodule_density_gt
        excel_temp['长径inf(mm)'] = self.long_inf
        excel_temp['短径inf(mm)'] = self.short_inf
        excel_temp['病灶体积inf(mm)'] = self.volume_inf
        # excel_temp['最大结节密度inf'] = self.max_nodule_density_inf
        excel_temp['平均结节密度inf'] = self.mean_nodule_density_inf
        excel_temp['分类仲裁结果'] = self.classification_results_gt
        excel_temp['分类检测结果'] = self.classification_results_inf
        excel_temp['预测检出概率'] = self.prob
        excel_temp['正常产品输出概率'] = self.normal_prob
        # excel_temp['最大结节密度'] = self.max_nodule_density
        # excel_temp['平均结节密度'] = self.mean_nodule_density
        excel_temp['检出统计结果'] = self.statistical_result
        excel_temp['交并比'] = self.IoU
        excel_temp['中心点距离'] = self.center_distance
        return excel_temp